import torch
from torch import nn, optim

data = torch.randn(128, 512, 3, 3 )
conv1 = nn.ConvTranspose2d(512, 256, 5, 3, padding=1, output_padding=2)

out = conv1(data)
print(out.shape)

a = torch.ones(3, 2)
data = torch.full_like(a, 1)
print(data)